__author__ = 'chobit'
# -*- coding:utf-8 -*-

import urllib
import urllib2
import tool
import re
import os


class Spider:
	def __init__(self):
		self.siteURL = 'http://mm.taobao.com/json/request_top_list.htm'
		self.too = tool.Tool()
	
	#获取索引页面内容
	def getPage(self,pageIndex):
		url = self.siteURL + "?page=" + str(pageIndex)
		print url
		request = urllib2.Request(url)
		response = urllib2.urlopen(request)
		return response.read().decode('gbk')
	
	#获取索引界面所有MM的信息，List
	def getContens(self,pageIndex):
		page = self.getPage(pageIndex)
		pattern = re.compile('<div class="list-item".*?pic-word.*?<a href="(.*?)".*?<img src="(.*?)".*?<a class="lady-name.*?>(.*?)</a>.*?<strong>(.*?)</strong>.*?<span>(.*?)</span>',re.S)
		items = re.findall(pattern,page)
		contents = []
		for item in items:
			contents.append([item[0],item[1],item[2],item[3],item[4]])
			return contents

	#获取MM个人详情页面
	def getDetailPage(self,infoURL):
		reponse = urllib2.urlopen(infoURL)
		return response.read().decode('gbk')

	#获取个人文字简介
	def getBrief(self,page):
		pattern = re.compile('<div class="mm-aixiu-content".*?>(.*?)<!--',re.S)
		result = re.search(pattern,page)
		return self.tool.replace(result.group(1))

	#获取页面所有图片
	def getAllImg(self,page):
		pattern = re.compile('<div class="mm-aixiu-content".*?>(.*?)<!--',re.S)
		content = re.search(pattern,page)
		patternImg = re.compile('<img.*?src="(.*?)"',re.S)
		images = re.findall(patternImg,content.group(1))
		return images

	#保存多张写真图片
	def saveImages(self,images,name):
		number = 1
		print u'发现',name,u'共有',len(images),u'张照片'
		for imageURL in images:
			splitPath = imageURL.split('.')
			fTail = splitPath.pop()
			if len(fTail) > 3:
				fTail = 'jpg'
			fileName = name +'/' + str(number) + '.' + fTail
			self.saveImg(imageURL,fileName)
			number +=1

	#保存头像
	def saveIcon(self,iconURL,name):
		splitPath = iconURL.split('.')
		fTail = splitPath.pop()
		fileName = name +'/icon'+fTail
		self.saveImg(iconURL,fileName)

	#保存个人简介
	def saveBrief(self,content,name):
		fileName = name + "/" + name + ".txt"
		f = open(fileName,"w+")
		print u"正在偷偷保存她的个人信息为",fileName
		f.write(content.encode('utf-8'))
 
 	#传入图片地址,文件名，保存图片
	def saveImg(self,imageURL,fileName):
		urllib.urlretrieve(imageURL,'%s.jpg' % fileName)
		print u"正在悄悄保存她的一张图片为",fileName	
	#创建新目录
	def mkdir(self,path):
		path = path.strip()
		#判断路径是否存在
		#存在   	True
		#不存在 	Flase
		isExists = os.path.exists(path)
		#判断结构
		if not isExists:
			#不存在则创建目录
     		# print u"偷偷新建了名字叫做",path,u'的文件夹'
			os.makedirs(path)
			return True
		else:
			#存在则不创建目录，
			print u"名为",path,'的文件夹已经创建成功'	
			return False

	#将第一页的淘宝MM的信息保存下载
	def savePageInfo(self,pageIndex):
		        #获取第一页淘宝MM列表
        contents = self.getContents(pageIndex)
        for item in contents:
            #item[0]个人详情URL,item[1]头像URL,item[2]姓名,item[3]年龄,item[4]居住地
            print u"发现一位模特,名字叫",item[2],u"芳龄",item[3],u",她在",item[4]
            print u"正在偷偷地保存",item[2],"的信息"
            print u"又意外地发现她的个人地址是",item[0]
            #个人详情页面的URL
            detailURL = item[0]
            #得到个人详情页面代码
            detailPage = self.getDetailPage(detailURL)
            #获取个人简介
            brief = self.getBrief(detailPage)
            #获取所有图片列表
            images = self.getAllImg(detailPage)
            self.mkdir(item[2])
            #保存个人简介
            self.saveBrief(brief,item[2])
            #保存头像
            self.saveIcon(item[1],item[2])
            #保存图片
            self.saveImgs(images,item[2])
 
 	#传入开始页面，获取MM图片
 	def savePagesInfo(self,start,end):
 		for i in range(start,end+1):
 	    	print u"正在偷偷寻找第",i,u"个地方，看看MM们在不在"
            self.savePageInfo(i)



spider = Spider()
spider.savePagesInfo(2,10)